/*
 * Tencent is pleased to support the open source community by making spring-cloud-tencent available.
 *
 * Copyright (C) 2021 Tencent. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * Unless required by applicable law or agreed to in writing, software distributed
 * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 * CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package com.tencent.cloud.metadata.core;

import java.util.Map;

import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletRequestWrapper;


public class HttpServletRequestHeaderWrapper extends HttpServletRequestWrapper {

	private  Map<String, String> addHeaders;

	public HttpServletRequestHeaderWrapper(HttpServletRequest request,  Map<String, String> addHeaders) {
		super(request);
		this.addHeaders = addHeaders;
	}


	@Override
	public String getHeader(String name) {
		if (addHeaders.containsKey(name)) {
			return addHeaders.get(name);
		}
		else  {
			return super.getHeader(name);
		}
	}
}
